#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Copyright (c) 2025 粤港澳大湾区（广东）国创中心

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <https://www.gnu.org/licenses/>.

File: pyncti_mesh_add_gravity.py
Desc: Function to add gravity to selected mesh entities.
Author: xumanjia
Date: 2025-08-01
"""
# 选中整个非结构化网络，根据设置的方向和强度设置重力
def pyncti_mesh_add_gravity(NCTI,doc):
    doc.ResetCaseResult()
    sel = NCTI.SelectionManager(doc)
    if len(sel.ObjectNames) > 0: 
        vec = NCTI.Vector(0, 0, -1)
        strength = 45
        res  = doc.ReturnDialogData(-1, "添加重力参数", '方向',vec, '强度', strength)
        if len(res) > 0:
            doc.RunCommand("cmd_ncti_mesh_add_gravity", "graviry", sel.ObjectNames[0], res[0], res[1])
        else:
            print('Cancle')
    else:
        print("没有选择数据")